package com.qf.mybatisplus.config.rabbitMQConfig;

import org.springframework.amqp.core.Message;
import org.springframework.amqp.rabbit.connection.CorrelationData;
import org.springframework.amqp.rabbit.core.RabbitTemplate;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Component;

import javax.annotation.PostConstruct;

//@Component
public class PublisherConfirmAndReturnConfig implements RabbitTemplate.ConfirmCallback, RabbitTemplate.ReturnCallback {

    @Autowired
    private RabbitTemplate rabbitTemplate;

    // 把 confirm 和 return 设置到rabbitTemplate里面
    @PostConstruct  // init-method
    public void initMethod(){
        rabbitTemplate.setConfirmCallback(this);
        rabbitTemplate.setReturnCallback(this);
    }


    /**
     * confirm 机制回调函数
     * @param correlationData
     * @param ack  消息是否到达 exchange
     * @param cause 消息没有到达 exchange 原因
     */
    @Override
    public void confirm(CorrelationData correlationData, boolean ack, String cause) {

        if(ack){
            System.err.println("消息已经送达到Exchange");
        }else {
            System.err.println("消息没有送达到Exchange"+cause);
        }
    }

    /**
     * 消息没有到达 Queue 才会执行此方法
     * @param message 消息对象
     * @param replyCode
     * @param replyText
     * @param exchange
     * @param routingKey
     */
    @Override
    public void returnedMessage(Message message, int replyCode, String replyText, String exchange, String routingKey) {

        System.err.println("消息没有到达Queue");
    }
}
